1import type { BundledLanguage } from "shiki";
2
3export function inferLanguage(filePath: string): BundledLanguage | undefined {
4  const extension = filePath.split(".").pop()?.toLowerCase();
5  const fileName = filePath.split("/").pop()?.toLowerCase();
6
7  if (fileName === "dockerfile") return "dockerfile";
8  if (fileName === "makefile") return "makefile";
9  if (fileName === "codeowners") return "codeowners";
10  if (fileName === ".env" || fileName?.startsWith(".env.")) return "dotenv";
11
12  const extensionMap: Record<string, BundledLanguage> = {
13    ts: "typescript",
14    tsx: "tsx",
15    js: "javascript",
16    jsx: "jsx",
17    mjs: "mjs",
18    cjs: "cjs",
19    py: "python",
20    rs: "rust",
21    go: "go",
22    java: "java",
23    c: "c",
24    cpp: "cpp",
25    cc: "cpp",
26    cxx: "cpp",
27    h: "c",
28    hpp: "cpp",
29    cs: "csharp",
30    rb: "ruby",
31    php: "php",
32    swift: "swift",
33    kt: "kotlin",
34    kts: "kts",
35    scala: "scala",
36    r: "r",
37    dart: "dart",
38    lua: "lua",
39    sql: "sql",
40    html: "html",
41    css: "css",
42    scss: "scss",
43    sass: "sass",
44    less: "less",
45    json: "json",
46    jsonc: "jsonc",
47    json5: "json5",
48    yaml: "yaml",
49    yml: "yml",
50    toml: "toml",
51    xml: "xml",
52    md: "markdown",
53    mdx: "mdx",
54    sh: "bash",
55    bash: "bash",
56    zsh: "zsh",
57    fish: "fish",
58    ps1: "powershell",
59    bat: "batch",
60    cmd: "cmd",
61    vue: "vue",
62    svelte: "svelte",
63    astro: "astro",
64    elm: "elm",
65    erl: "erlang",
66    ex: "elixir",
67    exs: "elixir",
68    fs: "fsharp",
69    hs: "haskell",
70    clj: "clojure",
71    coffee: "coffeescript",
72    nim: "nim",
73    v: "v",
74    zig: "zig",
75    graphql: "graphql",
76    gql: "graphql",
77    proto: "protobuf",
78    tf: "terraform",
79    tfvars: "tfvars",
80    hcl: "hcl",
81    dockerfile: "dockerfile",
82    tex: "latex",
83    vim: "vim",
84    asm: "asm",
85    sol: "solidity",
86    vy: "vyper",
87    move: "move",
88    cairo: "cairo",
89    prisma: "prisma",
90    adoc: "asciidoc",
91    rst: "rst",
92    diff: "diff",
93    csv: "csv",
94    tsv: "tsv",
95  };
96
97  return extension ? extensionMap[extension] : undefined;
98}
99